【Python】 循环遍历Numpy中的Array

您所在的位置:网站首页 python for循环 数组代码 【Python】 循环遍历Numpy中的Array

【Python】 循环遍历Numpy中的Array

2024-07-12 14:44| 来源: 网络整理| 查看: 265

1. 引言

Numpy是Python中常见的数据处理库。Numpy是 Numerical Python的缩写,它是数据科学中经常使用的库。Numpy专门用于处理矩阵运算,因为它包含各式各样的处理函数。在本文中,我们主要用于学习如何迭代遍历访问矩阵中的元素。

闲话少说,我们直接开始吧!

2. 使用For循环遍历

首先我们来看个例子,使用循环来遍历数组,样例代码如下:

import numpy as np array = np.array([1, 2, 3, 4, 5, 6]) for x in array: print(x) Output: 1 2 3 4 5 6

在上面的例子中,我们创建了一个一维数组,并成功地遍历访问了每个值。现在让我们来看一个二维矩阵中的例子:

import numpy as np array = np.array([[1, 2, 3], [4, 5, 6]]) for x in array: for y in x: print(y) Output: 1 2 3 4 5 6

正如我们在上面例子中所看到的,我们仍然能够打印出每个单独的值。因为它是一个二维数组,所以我们必须使用两个for循环来输出每个单独的值。这是我们通常迭代二维数组的方式,但NumPy为我们提供了新的函数,使得迭代NumPy数组变得更容易。

3. 函数 nditer()

函数nditer()主要用于循环遍历整个数组,而无需为每个额外维度使用嵌套for循环。 我们不妨来看一个例子:

import numpy as np array = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) for x in np.nditer(array): print(x) Output: 1 2 3 4 5 6 7 8 9

上述例子是一个二维的数组,我们使用函数nditer()后,我们不需要再使用嵌套的for循环。函数‘nditer()’成功地f访问并打印了数组中的每个值。 我们不妨在来看一个三维数组的例子,样例如下:

import numpy as np array = np.array([[[1], [2]], [[3], [4]]]) for x in np.nditer(array): print(x) Output: 1 2 3 4

正如我们在上面的例子中所看到的,函数nditer()成功地迭代了三维数组中的每个元素。

4. 函数 ndenumerate()

接着我们来介绍函数ndenumerate(),该函数的作用是输出相应的索引号的对应的值。 样例代码如下:

import numpy as np array = np.array([1, 2, 3, 4, 5, 6]) for i, x in np.ndenumerate(array): print(i, x) Output: (0,) 1 (1,) 2 (2,) 3 (3,) 4 (4,) 5 (5,) 6

正如上述例子中,我们在括号内输出了每个元素的索引号及其相应的值。接着我们再来看一个二维矩阵的例子:

import numpy as np array = np.array([[1, 2, 3], [4, 5, 6]]) for i, x in np.ndenumerate(array): print(i, x) Output: (0, 0) 1 (0, 1) 2 (0, 2) 3 (1, 0) 4 (1, 1) 5 (1, 2) 6

在上述例子中,我们输出的第一维表示每个元素的索引号,第二维表示每个元素的值。

5. 结论

本文重点介绍了在Numpy中常用的两个函数nditer()以及ndenumerate(),这两个函数在高维数组中循环遍历时非常有用,希望大家可以在日常工作中多多使用。

在这里插入图片描述 关注公众号《AI算法之道》,获取更多AI算法资讯。

在这里插入图片描述



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3